import paramiko, os
from wxauto import WeChat

# 创建SSHClient实例
client = paramiko.SSHClient()

# 允许连接不在know_hosts文件中的主机
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接服务器
client.connect('10.118.7.54', username='fszc', password='Zxm,10***')

# 打开SFTP客户端
sftp = client.open_sftp()

# 下载文件
sftp.get('/home/fszc/DJ-CLOUD/gateway/logs/js-gateway/error.log',
         r'C:\Users\00\Desktop\error.log')
sftp.get('/home/fszc/DJ-CLOUD/gateway/logs/js-gateway/info.log',
         r'C:\Users\00\Desktop\info.log')

# 关闭连接
sftp.close()

# 关闭连接
client.close()
print('日志下载完成')


# 发送日志文件到微信
wx = WeChat()
# 发送日志文件到微信
files = [r'C:\Users\00\Desktop\info.log', r'C:\Users\00\Desktop\error.log']
who = '动力健康度通报'
wx.SendFiles(filepath=files, who=who)
# 发送消息给文件传输助手  动力健康度通报
msg = '今日巡检服务器，查询日志无异常登录及异常操作记录情况。服务器日志已备份。'
wx.SendMsg(msg=msg, who=who)
for file in files:
    os.remove(file)
